package it.storm.solution;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 697. 数组的度
 * https://leetcode-cn.com/problems/degree-of-an-array/
 */
public class Solutions_697 {
    public static void main(String[] args) {
        int[] nums1 = {1, 2, 2, 3, 1};  // output: 2
        int[] nums2 = {1, 2, 2, 3, 1, 4, 2};  // output: 6
        List<int[]> inputs = new ArrayList<>();
        inputs.add(nums1);
        inputs.add(nums2);

        for (int[] input : inputs) {
            int result = findShortestSubArray(input);
            System.out.println(result);
        }
    }

    /**
     * 解法二：哈希表（数组方式实现）（2ms）
     */
    public static int findShortestSubArray(int[] nums) {
        if (nums == null) {
            return 0;
        }
        int min = Integer.MAX_VALUE;
        int max = Integer.MIN_VALUE;
        for (int num : nums) {
            if (num < min) {
                min = num;
            }
            if (num > max) {
                max = num;
            }
        }
        int[] counts = new int[max - min + 1];
        int degree = 0;
        for (int num : nums) {
            // 统计次数同时比较得到最大次数
            degree = Math.max(degree, ++ counts[num - min]);
        }
        if (degree == 1) {
            return 1;
        }

        int res = Integer.MAX_VALUE;
        for (int i = 0; i < counts.length; i++) {
            if (counts[i] != degree) {
                continue;
            }
            // 计算元素 min + i 在 nums 中的最左最右索引
            int num = min + i;
            int start = 0;
            int end = nums.length - 1;
            while (start < end && nums[start] != num) {
                start ++;
            }
            while (end > start && nums[end] != num) {
                end --;
            }
            // [start, end] 区间就是包含了元素 num 的最短子数组
            res = Math.min(res, end - start + 1);
        }
        return res;
    }

    /**
     * 解法一：哈希表（32ms）
     */
    public static int findShortestSubArray2(int[] nums) {
        if (nums == null) {
            return 0;
        }
        // 记录下每个元素出现的次数
        Map<Integer, Integer> map = new HashMap<>();
        // 记录每个元素第一次出现与最后一次出现的位置
        Map<Integer, int[]> locaMap = new HashMap<>();

        for (int i = 0; i < nums.length; i++) {
            map.put(nums[i], map.getOrDefault(nums[i], 0) + 1);
            if (!locaMap.containsKey(nums[i])) {
                locaMap.put(nums[i], new int[]{i, i});
            } else {
                locaMap.get(nums[i])[1] = i;
            }
        }

        int max = 0;
        // 记录下出现次数最多的元素（可能有多个）
        List<Integer> list = new ArrayList<>();
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (entry.getValue() < max) {
                continue;
            }
            if (entry.getValue() > max) {
                max = entry.getValue();
                list.clear();
            }
            list.add(entry.getKey());
        }

        int res = nums.length + 1;
        // 比较得到最短子数组
        for (int num : list) {
            int[] location = locaMap.get(num);
            int cur = location[1] - location[0] + 1;
            if (cur < res) {
                res = cur;
            }
        }
        return res;
    }
}
